package com.kylin.system.service;

import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.kylin.common.core.exception.CheckException;
import com.kylin.common.core.exception.ServiceException;
import com.kylin.common.core.service.impl.DataServiceImpl;
import com.kylin.common.redis.service.RedisService;
import com.kylin.system.common.enums.SystemRedisKeys;
import com.kylin.system.entity.SysRole;
import com.kylin.system.entity.SysRoleDept;
import com.kylin.system.entity.SysRoleMenu;
import com.kylin.system.mapper.SysRoleDeptMapper;
import com.kylin.system.mapper.SysRoleMapper;
import com.kylin.system.mapper.SysRoleMenuMapper;
import com.kylin.system.mapper.SysUserRoleMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.*;

/**
 * 角色 业务层处理
 * 
 * @author wuhao
 */
@Service
public class SysRoleService extends DataServiceImpl<SysRoleMapper, SysRole>
{
    @Resource
    private SysRoleMapper dao1;

    @Resource
    private SysRoleMenuMapper roleMenuMapper;

    @Resource
    private SysUserRoleMapper userRoleMapper;

    @Resource
    private SysRoleDeptMapper roleDeptMapper;

    @Autowired
    private RedisService redisService;


    /**
     * 根据用户ID查询权限
     * 
     * @param userId 用户ID
     * @return 权限列表
     */
    public Set<String> selectRolePermissionByUserId(Long userId)
    {
        List<SysRole> perms = dao1.selectRolePermissionByUserId(userId);
        Set<String> permsSet = new HashSet<>();
        for (SysRole perm : perms)
        {
            if (ObjectUtils.isNotEmpty(perm))
            {
                permsSet.addAll(Arrays.asList(perm.getRoleKey().trim().split(",")));
            }
        }
        return permsSet;
    }

    /**
     * 根据用户ID获取角色选择框列表
     * 
     * @param userId 用户ID
     * @return 选中角色ID列表
     */
    public List<Long> selectRoleListByUserId(Long userId)
    {
        return dao1.selectRoleListByUserId(userId);
    }


    /**
     * 校验角色权限是否唯一
     * 
     * @param role 角色信息
     * @return 结果
     */
    public void checkRoleKeyUnique(SysRole role)
    {
        Long roleId = ObjectUtils.isEmpty(role.getId()) ? -1L : role.getId();
        SysRole where = new SysRole();
        where.getQueryWrapper().eq("role_key", role.getRoleKey());
        SysRole info = dao1.selectOne(where.getQueryWrapper());
        if (ObjectUtils.isNotEmpty(info) && info.getId().longValue() != roleId.longValue())
        {
            throw  new CheckException("新增角色'" + role.getRoleName() + "'失败，角色权限已存在");
        }
    }

    /**
     * 校验角色是否允许操作
     * 
     * @param role 角色信息
     */
    public void checkRoleAllowed(SysRole role)
    {
        if (ObjectUtils.isNotEmpty(role.getId()) && role.isAdmin())
        {
            throw new ServiceException("不允许操作超级管理员角色");
        }
    }

    /**
     * 通过角色ID查询角色使用数量
     * 
     * @param roleId 角色ID
     * @return 结果
     */
    public int countUserRoleByRoleId(Long roleId)
    {
        return userRoleMapper.countUserRoleByRoleId(roleId);
    }

    /**
     * 新增保存角色信息
     * 
     * @param role 角色信息
     * @return 结果
     */
    @Transactional
    public int insertRole(SysRole role) {
        // 新增角色信息
        role.perInsert();
        // 清除缓存信息
        this.clearCache();
        return dao1.insert(role);
    }

    /**
     * 修改保存角色信息
     * 
     * @param role 角色信息
     * @return 结果
     */
    @Transactional
    public int updateRole(SysRole role)
    {
        // 修改角色信息
        role.perUpdate();
        // 清除缓存信息
        this.clearCache();

        return dao1.updateById(role);
    }

    /**
     * 修改数据权限信息
     * 
     * @param role 角色信息
     * @return 结果
     */
    @Transactional
    public int authDataScope(SysRole role)
    {
        // 修改角色信息
        dao1.updateById(role);
        // 删除角色与部门关联
        roleDeptMapper.deleteRoleDeptByRoleId(role.getId());

        // 清除缓存信息
        this.clearCache();

        // 新增角色和部门信息（数据权限）
        return insertRoleDept(role);
    }

    /**
     * 修改菜单权限
     * @param role 角色信息
     * @return
     * @author wuhao
     * @version 1.0.0 - 2024/10/23
     */
    @Transactional
    public int authMenus(SysRole role) {
        // 修改角色信息
        dao1.updateById(role);
        // 删除角色与菜单关联
        roleMenuMapper.deleteRoleMenuByRoleId(role.getId());

        // 清除缓存信息
        this.clearCache();

        return insertRoleMenu(role);
    }

    /**
     * 新增角色菜单信息
     * 
     * @param role 角色对象
     */
    public int insertRoleMenu(SysRole role)
    {
        int rows = 0;
        // 新增用户与角色管理
        if(role.getMenuIds() != null) {
            List<SysRoleMenu> list = new ArrayList<>();
            for (Long menuId : role.getMenuIds()) {
                SysRoleMenu rm = new SysRoleMenu();
                rm.setRoleId(role.getId());
                rm.setMenuId(menuId);
                list.add(rm);
            }
            if (!list.isEmpty()) {
                rows = roleMenuMapper.batchRoleMenu(list);
            }
        }
        // 清除缓存信息
        this.clearCache();
        return rows;
    }

    /**
     * 新增角色部门信息(数据权限)
     *
     * @param role 角色对象
     */
    public int insertRoleDept(SysRole role)
    {
        int rows = 1;
        // 新增角色与部门（数据权限）管理
        List<SysRoleDept> list = new ArrayList<>();
        if(role.getDeptIds() != null) {
            for (Long deptId : role.getDeptIds()) {
                SysRoleDept rd = new SysRoleDept();
                rd.setRoleId(role.getId());
                rd.setDeptId(deptId);
                list.add(rd);
            }
            if (!list.isEmpty()) {
                rows = roleDeptMapper.batchRoleDept(list);
            }
        }

        // 清除缓存信息
        this.clearCache();
        return rows;
    }

    /**
     * 批量删除角色信息
     * 
     * @param roleIds 需要删除的角色ID
     * @return 结果
     */
    public int deleteRoleByIds(Long[] roleIds)
    {
        for (Long roleId : roleIds)
        {
            checkRoleAllowed(new SysRole(roleId));
            SysRole role = getById(roleId);
            if (countUserRoleByRoleId(roleId) > 0)
            {
                throw new ServiceException(String.format("%1$s已分配,不能删除", role.getRoleName()));
            }
        }
        int count  = dao1.deleteBatchIds(Arrays.asList(roleIds));
        // 清除缓存信息
        this.clearCache();
        return count;
    }

    /**
     * 清除缓存信息
     * @author wuhao
     * @version 1.0.0 - 2023/4/8
     */
    @Override
    public void clearCache() {
        redisService.deleteObject(SystemRedisKeys.SYS_ROLE_KEY.key());
    }
}
